package com.example.springai.controller;

import cn.hutool.core.util.StrUtil;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.example.springai.common.R;
import com.example.springai.domain.AiDocument;
import com.example.springai.service.IAiDocumentService;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.util.List;

/**
 * AI知识库控制器
 *
 * @author ZWX
 */
@RestController
@RequestMapping("/ai/document")
@RequiredArgsConstructor
public class AiDocumentController {
    private final IAiDocumentService aiDocumentService;

    /**
     * 上传文件
     *
     * @param file 文件
     * @return 上传结果
     */
    @PostMapping("/upload")
    public R upload(@RequestParam("file") MultipartFile file) {
        return aiDocumentService.uploadDocument(file) ? R.ok() : R.fail();
    }

    /**
     * 获取文件列表
     *
     * @param req 请求数据
     * @return 文件列表
     */
    @GetMapping("/list")
    public R<List<AiDocument>> list(AiDocument req) {
        return R.ok(aiDocumentService.getList(req));
    }

    /**
     * 删除文件
     *
     * @param documentId 文件ID
     * @return 删除结果
     */
    @DeleteMapping("/delete/{documentId}")
    public R delete(@PathVariable("documentId") Long documentId) {
        return aiDocumentService.deleteDocument(documentId) ? R.ok() : R.fail();
    }

    /**
     * 根据文档ID获取文档详情
     *
     * @param documentId 文档ID
     * @return 文档详情（包含文档块列表）
     */
    @GetMapping("/detail/{documentId}")
    public R<AiDocument> getDetail(@PathVariable("documentId") Long documentId) {
        AiDocument documentDetail = aiDocumentService.getDocumentDetail(documentId);
        return documentDetail != null ? R.ok(documentDetail) : R.fail();
    }

}
